import torch
#pip uninstall torch torchvision torchaudio
#pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
#pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# 检查PyTorch是否有CUDA支持
print(torch.cuda.is_available())  # 应该返回True

# 检查当前GPU设备数量
print(torch.cuda.device_count())  # 应该大于0

# 检查当前GPU名称
print(torch.cuda.get_device_name(0))  # 应该显示你的GPU型号

# 检查PyTorch版本和CUDA版本
print(torch.__version__)
print(torch.version.cuda)  # 应该显示与你的CUDA工具包匹配的版本 pip config list